# CsGen2-train:
CUDA_VISIBLE_DEVICES=6 python -m tasks.ClassSummary.models.CsGen2.main \
  --save_path ./model/CsGen2/ \
  --train ./tasks/ClassSummary/resource/train.token.jsonl \
  --val ./tasks/ClassSummary/resource/valid.token.jsonl \
  --units_field ./tasks/ClassSummary/resource/units.field \
  --csbt_field ./tasks/ClassSummary/resource/csbt.field \
  --nl_field ./tasks/ClassSummary/resource/nl.field \
  --patience 4 \
  --batch_size 64 \
  --max_epoch 100 \
  --embedding 128 \
  --hidden 256 \
  --layers 2 \
  --lr 0.001 \
  --max_grad 5 \
  --gpu_num 1 \
  --seed 1024 \
  --check_freq 1

# CsGen2-test-hybrid:
CUDA_VISIBLE_DEVICES=5 python -m tasks.ClassSummary.models.CsGen2.test \
  --test ./tasks/ClassSummary/resource/test.token.jsonl \
  --code_field ./tasks/ClassSummary/resource/code.field \
  --nl_field ./tasks/ClassSummary/resource/nl.field \
  --check_point /home/hlcs/lmc/clsum/model/CsGen2/CsGen-epoch=27-bleu=12.76.ckpt \
  --save_path ./translate/CsGen2-hybrid.jsonl \
  --gpu_num 1 \
  --batch_size 64 \
  --seed 1024

# CsGen2-test-cross:
CUDA_VISIBLE_DEVICES=5 python -m tasks.ClassSummary.models.CsGen2.test \
  --test ./tasks/ClassSummary/resource/cross.token.jsonl \
  --code_field ./tasks/ClassSummary/resource/code.field \
  --nl_field ./tasks/ClassSummary/resource/nl.field \
  --check_point /home/hlcs/lmc/clsum/model/CsGen2/CsGen-epoch=27-bleu=12.76.ckpt \
  --save_path ./translate/CsGen2-cross.jsonl \
  --gpu_num 1 \
  --batch_size 64 \
  --seed 1024
